import numpy as np
from scipy.cluster.vq import kmeans, vq
from demo import show_plt

# 创建一个示例数据集
data = np.array([[2, 5], [1, 4], [3, 6], [8, 2], [7, 3], [6, 4]], dtype=np.float32)
# 指定簇的数量
k = 2

# 使用kmeans函数进行聚类
centroids, labels = kmeans(data, k)

# 使用vq函数将样本分配到对应的簇
labels, _ = vq(data, centroids)

print(labels)
show_plt(data, centroids)
